package com.hongyi.common.mybatis.mapper;

import com.baomidou.mybatisplus.core.conditions.Wrapper;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.mapper.BaseMapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.core.toolkit.reflect.GenericTypeUtils;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.toolkit.Db;
import com.hongyi.common.framework.utils.common.CollectionUtils;
import com.hongyi.common.framework.utils.bean.MapstructUtils;
import com.hongyi.common.framework.utils.common.ObjectUtils;

import java.io.Serializable;
import java.util.ArrayList;
import java.util.Collection;
import java.util.List;
import java.util.Map;

public interface BaseMapperExtend<T,V> extends BaseMapper<T> {

    /**
     * 批量插入
     *
     * @param entityList 需要批量插入的实体
     */
    default boolean insertBatch(Collection<T> entityList) {
        return Db.saveBatch(entityList);
    }

    /**
     * 批量插入(包含限制条数)
     *
     * @param entityList 需要批量插入的实体
     * @param batchSize 限制条数
     */
    default boolean insertBatch(Collection<T> entityList, int batchSize) {
        return Db.saveBatch(entityList, batchSize);
    }

    /**
     * 批量更新
     *
     * @param entityList 需要批量更新的实体
     */
    default boolean updateBatchById(Collection<T> entityList) {
        return Db.updateBatchById(entityList);
    }

    /**
     * 批量更新(包含限制条数)
     * @param entityList 需要批量更新的实体
     * @param batchSize 限制条数
     */
    default boolean updateBatchById(Collection<T> entityList, int batchSize) {
        return Db.updateBatchById(entityList, batchSize);
    }

    /**
     * 通过泛型获取指定的VO类
     */
    @SuppressWarnings("unchecked")//忽略编译器警告的注解
    default Class<V> currentVoClass() {
        return (Class<V>) GenericTypeUtils.resolveTypeArguments(this.getClass(), BaseMapperExtend.class)[1];
    }

    /**
     * 查询实体类全部记录
     */
    default List<T> selectList(){
        return this.selectList(new QueryWrapper<>());
    }

    /**
     * 查询VO列表全部记录
     */
    default List<V> selectVoList() {
        return selectVoList(new QueryWrapper<>(),this.currentVoClass());
    }

    /**
     * 根据条件查询VO列表记录
     *
     * @param queryWrapper 查询条件
     */
    default List<V> selectVoList(Wrapper<T> queryWrapper ) {
        return selectVoList(queryWrapper,this.currentVoClass());
    }

    /**
     * 根据 entity 条件，查询全部记录
     *
     * @param queryWrapper 查询条件
     * @param  voClass 需要查询的Vo类
     */
    default <C> List<C> selectVoList(Wrapper<T> queryWrapper, Class<C> voClass) {
        List<T> queryList = this.selectList(queryWrapper);
        if (CollectionUtils.isEmpty(queryList)) {
            return new ArrayList<>();
        }
        return MapstructUtils.convert(queryList,voClass);
    }

    /**
     * 根据ID查询自定义Vo类
     *
     * @param id 查询主键
     * @param  voClass 需要查询的Vo类
     */
    default <C>C selectVoById(Serializable id, Class<C> voClass){
        T object = this.selectById(id);
        if (ObjectUtils.isNull(object)) {
            return null;
        }
        return MapstructUtils.convert(object,voClass);
    }

    /**
     * 根据ID查询指定Vo类
     *
     * @param id 查询主键
     */
    default V selectVoById(Serializable id) {
        return this.selectVoById(id,currentVoClass());
    }

    /**
     * 根据ID 批量查询指定Vo类
     *
     * @param idList ID集合
     */
    default List<V> selectVoBatchIds(Collection<? extends Serializable> idList) {
        return selectVoBatchIds(idList, this.currentVoClass());
    }

    /**
     * 根据ID 批量查询自定义Vo类
     *
     * @param idList ID集合
     * @param  voClass 需要查询的Vo类
     */
    default <C> List<C> selectVoBatchIds(Collection<? extends Serializable> idList, Class<C> voClass) {
        List<T> list = this.selectBatchIds(idList);
        if (CollectionUtils.isEmpty(list)) {
            return CollectionUtils.newArrayList();
        }
        return MapstructUtils.convert(list, voClass);
    }

    /**
     * 指定当前VO的分页查询
     *
     * @param page 分页数据
     * @param queryWrapper 查询条件
     */
    default <P extends IPage<V>> P selectVoPage(IPage<T>page, Wrapper<T> queryWrapper){
        return this.selectVoPage(page,queryWrapper,this.currentVoClass());
    }

    /**
     * 自定义VO的分页查询
     *
     *  @param page 分页数据
     *  @param queryWrapper 查询条件
     *  @param  voClass 需要查询的Vo类
     */
    default <C,P extends IPage<C>> P selectVoPage(IPage<T>page, Wrapper<T> queryWrapper, Class<C> voClass){
        List<T> queryList = this.selectList(page, queryWrapper);
        IPage<C> voPage = new Page<>(page.getCurrent(), page.getSize(), page.getTotal());
        if (CollectionUtils.isEmpty(queryList)) {
            return (P) voPage;
        }
        voPage.setRecords(MapstructUtils.convert(queryList,voClass));
        return (P) voPage;
    }

    /**
     * 根据 columnMap 条件查询自定义Vo类
     *
     * @param map 查询条件
     * @param  voClass 需要查询的Vo类
     */
    default <C> List<C> selectVoByMap(Map<String, Object> map, Class<C> voClass) {
        List<T> list = this.selectByMap(map);
        if (CollectionUtils.isEmpty(list)) {
            return CollectionUtils.newArrayList();
        }
        return MapstructUtils.convert(list, voClass);
    }

    /**
     * 根据 columnMap 条件查询指定VO类
     *
     * @param map 查询条件
     */
    default List<V> selectVoByMap(Map<String, Object> map) {
        return selectVoByMap(map, this.currentVoClass());
    }

    /**
     * 根据 entity 条件，查询当前VO的一条记录
     * @param queryWrapper 查询条件
     */
    default V selectVoOne(Wrapper<T> queryWrapper) {
        return selectVoOne(queryWrapper,this.currentVoClass());
    }

    /**
     * 根据 entity 条件，查询自定义VO的一条记录
     *
     * @param queryWrapper 查询条件
     * @param  voClass 需要查询的Vo类
     */
    default <C> C selectVoOne(Wrapper<T> queryWrapper, Class<C> voClass){
        T object = this.selectOne(queryWrapper);
        if (ObjectUtils.isNull(object)) {
            return null;
        }
        return MapstructUtils.convert(object,voClass);

    }

}
